
###
 # @Author: daniel
 # @Date: 2024-02-29 15:28:31
 # @LastEditTime: 2024-03-17 16:09:22
 # @LastEditors: daniel
 # @Description: 
 # @FilePath: /mtpsl/install.sh
 # have a nice day
### 


 conda  create -n mtpsl python=3.7 -y



conda activate mtpsl


# conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch
# conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.6 -c pytorch -c conda-forge
# conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

python -c "import torch; print(torch.cuda.is_available(),torch.version.cuda)"



pip install pytorch-lightning
pip install omegaconf
pip install einops
pip install progress 
pip install open_clip_torch



pip install numpy 
pip install chardet
pip install matplotlib 
pip install scipy
pip install loguru





#!=================================================================
#* not working
# pip install tensorboard==2.6.0
# pip install tensorboard-data-server==0.7.2
# pip install tensorboard-plugin-wit==1.6.post3
# pip install tensorboardX==1.8
#!=================================================================


pip install ipython
pip install jupyter 

#* for train
pip install transformers
pip install open_clip_torch


pip install opencv-python

#* download pretrained model 
#* md5sum: output a secret code which is used to check the completion of the file. 
#* If the secret code is same as before, it is complete. 

pip install gdown
gdown https://drive.google.com/uc?id=1s9x8neT9SYR2M6C89CvbeID3XlBRJoEw && \
md5sum nyuv2_pretrained.zip && \
unzip nyuv2_pretrained.zip -d ./results/ && \
rm nyuv2_pretrained.zip


